/*
 * Copyright 2014 Gregor Schauer
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package at.dormancy.util;

import at.dormancy.metadata.ObjectMetadata;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.IdentityHashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Provides the context e.g., the metadata of the entities.
 *
 * @author Gregor Schauer
 * @since 2.0.0
 */
public class DormancyContext {
	final Map<Class<?>, ObjectMetadata> metadataMap = new ConcurrentHashMap<Class<?>, ObjectMetadata>();
	final Map<Object, Object> adjacencyMap = new IdentityHashMap<Object, Object>();

	public DormancyContext() {
	}

	public DormancyContext(@Nullable ObjectMetadata... metadata) {
		if (metadata != null) {
			for (ObjectMetadata m : metadata) {
				metadataMap.put(m.getType(), m);
			}
		}
	}

	/**
	 * Returns the metadata for the given object type.
	 *
	 * @param clazz the type of the object
	 * @return the metadata or {@code null} if none are provided
	 */
	@Nullable
	public ObjectMetadata getObjectMetadata(@Nonnull Class<?> clazz) {
		return metadataMap.get(clazz);
	}

	/**
	 * Returns a mapping of the transient objects that have already been visited and their persistent counterpart.
	 *
	 * @return the visited objects
	 */
	@Nonnull
	public Map<Object, Object> getAdjacencyMap() {
		return adjacencyMap;
	}
}
